sdk/samples/all in on code/Visual Studio 2008/CSHeapCorruption/Program.cs

   1 /******************************** Module Header ********************************\
   2 Module Name:  Program.cs
   3 Project:      CSHeapCorruption
   4 Copyright (c) Microsoft Corporation.
   5
   6
   7
   8 This source is subject to the Microsoft Public License.
   9 See http://www.microsoft.com/opensource/licenses.mspx#Ms-PL.
  10 All other rights reserved.
  11
  12 THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
  13 EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES
  14 OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
  15 \*******************************************************************************/
  16
  17 using System;
  18 using System.Runtime.InteropServices;
  19
  20
  21 namespace CSHeapCorruption
  22 {
  23     class Program
  24     {
  25         static void Main(string[] args)
  26         {
  27             if (args.Length > 0 && (args[0].StartsWith("-") || args[0].StartsWith("/")))
  28             {
  29                 string cmd = args[0].Substring(1);
  30
  31                 if (String.Compare(cmd, "o", true) == 0)
  32                 {
  33                     // Overrun the managed GC heap.
  34                     OverrunManagedGCHeap();
  35                 }
  36                 else if (String.Compare(cmd, "h", true) == 0)
  37                 {
  38                     // Mismatch the heap handle.
  39                     MismatchHeapHandle();
  40                 }
  41                 else
  42                 {
  43                     PrintInstructions();
  44                 }
  45             }
  46             else
  47             {
  48                 PrintInstructions();
  49             }
  50
  51             Console.Write("Press ENTER to exit ...");
  52             Console.ReadLine();
  53         }
  54
  55
  56         static void PrintInstructions()
  57         {
  58             Console.WriteLine("CSHeapCorruption Instructions:");
  59             Console.WriteLine("-o     Overrun the managed GC heap");
  60             Console.WriteLine("-h     Mismatch the heap handle");
  61         }
  62
  63
  64         #region Managed GC Heap Overrun
  65
  66         /// <summary>
  67         ///
  68         /// </summary>
  69         static void OverrunManagedGCHeap()
  70         {
  71             int[] buffer = new int[50];
  72             for (int i = 0; i < buffer.Length; i++)
  73             {
  74                 buffer[i] = 0xFF;
  75             }
  76
  77             Console.Write("Press ENTER to overrun GC heap ...");
  78             Console.ReadLine();
  79
  80             InitializeBuffer(buffer, buffer.Length);
  81
  82             //GC.Collect();
  83         }
  84
  85         [DllImport("CSHeapCorruption.NativeDll.dll", CharSet = CharSet.Unicode)]
  86         static extern void InitializeBuffer(int[] buffer, int size);
  87
  88         #endregion
  89
  90
  91         #region Heap Handle Mismatch
  92
  93         /// <summary>
  94         ///
  95         /// </summary>
  96         static void MismatchHeapHandle()
  97         {
  98             // Allocate a block of memory. The native function AllocateMemory
  99             // allocates the memory on the CRT heap.
 100             IntPtr pMem = AllocateMemory(50);
 101
 102             Console.Write("Press ENTER to mismatch heap handle ...");
 103             Console.ReadLine();
 104
 105             // Free the allocated memory on the wrong process heap.
 106             Marshal.FreeHGlobal(pMem);
 107         }
 108
 109         [DllImport("CSHeapCorruption.NativeDll.dll", CharSet = CharSet.Unicode)]
 110         static extern IntPtr AllocateMemory(int bytes);
 111
 112         #endregion
 113     }
 114 }